Developing a practical neurodevelopmental prediction model for targeting high-risk very preterm infants during visit after NICU: a retrospective national longitudinal cohort study

Background Follow-up visits for very preterm infants (VPI) after hospital discharge is crucial for their neurodevelopmental trajectories, but ensuring their attendance before 12 months corrected age (CA) remains a challenge. Current prediction models focus on future outcomes at discharge, but post-discharge data may enhance predictions of neurodevelopmental trajectories due to brain plasticity. Few studies in this field have utilized machine learning models to achieve this potential benefit with transparency, explainability, and transportability. Methods We developed four prediction models for cognitive or motor function at 24 months CA separately at each follow-up visits, two for the 6-month and two for the 12-month CA visits, using hospitalized and follow-up data of VPI from the Taiwan Premature Infant Follow-up Network from 2010 to 2017. Regression models were employed at 6 months CA, defined as a decline in The Bayley Scales of Infant Development 3rd edition (BSIDIII) composite score > 1 SD between 6- and 24-month CA. The delay models were developed at 12 months CA, defined as a BSIDIII composite score < 85 at 24 months CA. We used an evolutionary-derived machine learning method (EL-NDI) to develop models and compared them to those built by lasso regression, random forest, and support vector machine. Results One thousand two hundred forty-four VPI were in the developmental set and the two validation cohorts had 763 and 1347 VPI, respectively. EL-NDI used only 4–10 variables, while the others required 29 or more variables to achieve similar performance. For models at 6 months CA, the area under the receiver operating curve (AUC) of EL-NDI were 0.76–0.81(95% CI, 0.73–0.83) for cognitive regress with 4 variables and 0.79–0.83 (95% CI, 0.76–0.86) for motor regress with 4 variables. For models at 12 months CA, the AUC of EL-NDI were 0.75–0.78 (95% CI, 0.72–0.82) for cognitive delay with 10 variables and 0.73–0.82 (95% CI, 0.72–0.85) for motor delay with 4 variables. Conclusions Our EL-NDI demonstrated good performance using simpler, transparent, explainable models for clinical purpose. Implementing these models for VPI during follow-up visits may facilitate more informed discussions between parents and physicians and identify high-risk infants more effectively for early intervention. Supplementary Information The online version contains supplementary material available at 10.1186/s12916-024-03286-2.


Background
The recent progress in perinatal and postnatal care has contributed to enhancing mortality and morbidity outcomes among preterm infants for decades.However, approximately 20% of very preterm infants (VPI) survivors still suffer from a certain degree of cognitive or motor delay at 2 years of corrected age (CA) based on the Bayley score [1].The longitudinal follow-up program (LFUP) is regarded as the primary recommendation after neonatal intensive care unit (NICU) discharge for highrisk infants, particularly those born preterm during the first 2 years of life [2,3].Early detection and intervention of neurodevelopmental impairment (NDI) in high-risk infants can promote not only better outcomes but also social and economic benefits [4].Risk factors for NDI in preterm births, such as gestational age (GA), birth body weight (BBW), bronchopulmonary dysplasia, and intraventricular hemorrhage, have been well reported [5,6].
While evidence has substantiated the potential efficacy of innovative statistical tools for prediction models in this domain [7,8], it is noteworthy that existing prediction models predominantly focus on aiding healthcare practitioners and families during pre-discharge counseling [9,10].
High dropout rates exceeding 50% in LFUP make reliably evaluating VPI's development challenging.This uncertainty and developmental status changes may make parents think their child is improving and drop out of follow-up clinics before 1 year [11].Based on findings from the California low birthweight cohort study, early presence at a visit within the first 12 months emerged as the most significant determinant of sustained LFUP participation, alongside factors such as maternal education and proximity to the clinic [12].Notably, lack of awareness of early intervention is significantly related to attendance [13].
The accuracy and interpretability of neurodevelopmental prediction models influence caregivers' decisionmaking regarding counseling in clinics or hospitals [14,15].There remains a gap in prediction models for routine clinical use and NDI research [9].The machine learning methods have produced excellent results in prediction models for a variety of diseases [16], but complex machine learning models have been the subject of recent criticisms due to their lack of transparency.Furthermore, although the performance of simpler parametric models with lots of variables is not inferior to machine learning methods, even simple algorithms like logistic regression (LR) can become complicated by including numerous predictors [17,18].Recently, our novel evolutionary learning method was able to establish clinical prediction models by identifying a small set of features and maximizing the prediction accuracy [19,20].
This study aims to create practical predictive models at 6-month and 12-month visits for 24-month outcomes of CA with transparency, explainability, and transportability to help parents-physician discussion about early intervention and follow-up plans.

Study population
VPI was defined as preterm infants born before 31 weeks' 6 days of gestation weighing 401-1500 g. Between January 1, 2010, and December 31, 2014, 5615 VPI were discharged alive from 21 neonatal care centers registered in the Taiwan Premature Infant Follow-up Network (TPFN) database, comprising the original cohort [21,22].For establishing accurate prediction models, we excluded 3071 infants, including infants who missed the Bayley Scales of Infant Development 3rd edition (BSIDIII) at the 6 or 12 months, as they were key variables, and those whomissed BSIDIII scores at 2 years CA alone due to main predictive outcome, or any follow-up with the Bayley Scales of Infant Development 2nd edition.There were 2544 VPI with BSIDIII cognitive and motor scores at 2 years of CA in the development cohort.An external cohort consisting of 1347 VPI was obtained from the TPFN database from January 1, 2016, to December 31,2017, using the same criteria (Fig. 1).This study was approved by the ethical standards of the Institutional Review Board of the Kaohsiung Medical University Hospital (IRB number: KMUHIRB-SV(I)-20190008), and due to the study's retrospective nature and the use of deidentified data, the need for written informed consent was waived.

Neurodevelopmental outcome
Neurodevelopmental outcome in this study was based on BSIDIII score at 2 years of age of VPI.The TPFN followup plan included BSIDIII scores at 6, 12, and 24 months after CA by unblinded and experienced pediatric psychologists.Considering the BSIDIII score at 6 months may not be a reliable predictor of NDI at 24 months [23,24], we designed different NDI outcome models at 6-and 12-month CA for clinical use, comprising two regression models at 6 months and two delay models at 12 months.
Two delay models were used at 12 months: the cognitive delay model (CDelay) defined NDI as a BSIDIII cognitive score of < 85 at 24 months CA and the motor delay model (MDelay) defined NDI as a BSIDIII motor score < 85 at 24 months CA.Data for both models were collected up to 12 months after CA.Two regression models were used at 6 months: the cognitive regression (CRegres) defined NDI as a BSIDIII cognitive score decline greater than one standard deviation (SD) between 6 and 24 months CA and the motor regression (MRegres) model defined NDI as a BSIDIII motor score decline greater than one SD between 6 and 24 months CA, respectively.The data used in both regression models were up to 6-month CA, and the SD of BSIDIII composite score was 15 points in two regression models.

Prediction variables
The TPFN database contains basic demographics of parents, pregnancy, and neonatal variables at birth, hospitalization, discharge, and follow-ups at 6-, 12-, and 24-month CA.The anthropometric measures (weight, height, and head circumference) at four individual time points (admission, discharge, and 6-and 12-month CA) were redistributed into 8 intervals from less than − 3 Z to greater than 3 Z.We used the growth chart based on the INTERGROWTH-21st Project [25] at admission and discharge and the WHO Child Growth Standards [26] at the 6-and 12-month CA visit.Detailed definitions of all the variables are shown in Additional file 1.
A total of 484 variables in the TPFN database were obtained from preterm births to 12 months CA.First, we excluded variables with missing values in > 30% of the cohort, and there were 89 variables retained.We arranged missing data with imputation using the k-nearest neighbor method.However, we excluded six variables (peak bilirubin levels, blood transfusion, nasogastric tube feeding after discharge, apnea, partial pressure of oxygen, and carbon dioxide in the initial blood gas analysis) from consideration due to their absence in the external cohort dataset.Two distinct feature utilization strategies are employed in constructing prediction models.In light of the specific characteristics of random forest (RF) and lasso regression (Lasso) for handling a substantial number of predictors, an "all-features-in" approach is initially adopted for model development.Consequently, the CDelay and MDelay models are constructed with 83 variables, while the CRegres and MRegres models retain 75 variables for utilization in the Lasso and RF frameworks because the data used in regression models were only up to 6-month CA.
Second, we employ a Coarse-to-fine feature selection technique to facilitate model development, which applies to traditional machine learning methods and our novel approach.Coarse-to-fine feature selection from all recorded variables was performed as follows: Among 89 variables, the remaining 29 variables that were significantly related to NDI outcomes at 24 months based on Pearson's and Spearman's correlation coefficients were retained in a set of fine features for model development.The 29 variables for each prediction model and results of the correlation coefficients are shown in Additional file 2.

Evolutionary learning method
A novel evolutionary learning method, called evolutionary learning neurodevelopmental impairment (EL-NDI), was proposed to predict the NDI of VPI at 24 months of CA in this study.Figure 2 depicts the flowchart of developing EL-NDI.After excluding and including process in step 1, we divided the original cohort into development and independent test datasets at 7:3 in step 2. In the development datasets, each of the four NDI outcomes exhibited an imbalance.Subsequently, in step 3, we established four distinct balanced developmental cohorts.These cohorts were independently created by randomly pairing positive and negative cases at a 1:1 ratio drawn from the developmental dataset split in step 2. Consequently, each machine learning model was trained on a unique, balanced developmental cohort derived from the initially imbalanced development dataset, leading to variations in the cohort sizes.The 29-candidate features were obtained from the maternal, neonatal, and follow-up data with imputation using the k-nearest neighbor method.The predictive approach EL-NDI employed a widely recognized support vector machine (SVM) classifier, a statistically grounded supervised learning model.SVM are employed for classification or regression tasks through data transformation into a higher-dimensional feature space using a kernel function.The selection of both the cost parameter (C) and the kernel parameter (γ) in SVM is critical for modeling.We employed an intelligent evolutionary algorithm (IEA) [19] to determine SVM's optimal feature selection and parameter settings.The process involved the use of the inheritable bi-objective combinatorial genetic algorithm (IBCGA) [27] in conjunction with IEA to identify a subset of features and to determine the values of the SVM parameters while maximizing the Fig. 2 Illustrated flowchart of developing EL-NDI to predict neurodevelopmental impairment.EL-NDI utilized the inheritable bi-objective combinatorial genetic algorithm (IBCGA) alongside intelligent evolutionary algorithm (IEA) to identify feature subsets and optimize SVM parameters for maximum fitness.RF, random forest; LR, logistic regression; SVM, support vector machine fitness function.The fitness function aimed to maximize the prediction function of the tenfold cross-validation (CV) on the training dataset.The optimal feature selection problem, denoted as C(n, m), entails the selection of a small subset of features (m) from a more extensive set of candidates (n), where interactions among features exist.IBCGA was employed to efficiently address this large-scale combinatorial optimization challenge to determine the value of m, the selected features, and the values of C and γ.For the application of IBCGA, all candidate features were encoded as binary variables for optimal feature selection.Simultaneously, the parameters (C, γ) were encoded into the chromosome for concurrent optimization.Based on the main effect difference (MED), the selected m features were ranked according to their prediction contributions.For more information about the use of IBCGA, we recommend referring to our previously published biomedicine studies [28,29].

Models of machine learning
We used established machine learning models to compare the EL-NDI models.The models using the R package implementation included lasso regression, logistic regression (glmnet), linear SVM (e1071), and RF (ran-domForest).Additionally, we combined a small set of features selected by the EL-NDI with logistic regression as the evolutionary learning logistic regression (EL-LR) model to explore the relationship between the selected features and outcomes [9].After optimizing 19 hyperparameters for each model, we fitted the entire training set with five repetitions of tenfold cross-validation using the R caret package.

Statistical analysis
Statistical analyses were performed using R, version 3.6.3(R Foundation for Statistical Computing), Python, version 3.7 (Python Software Foundation), and MATLAB (version 2020a).A two-sided p ≤ 0.05 was considered statistically significant.We calculated 95% confidence intervals (CIs) to compare the area under the curve (AUC).Descriptive statistics were expressed as mean ± standard deviation or median (range) as appropriate.The Mann-Whitney U test was applied to compare continuous variables, while categorical variables were compared using Pearson χ 2 analysis or Fisher's exact test.

Characteristics of the cohorts
The attrition rates at 24 months CA in the original and external test cohorts were 38.9% and 18.7%, respectively.According to the study design, a total of 763 VPI from 2544 VPI in original cohorts were distributed to the independent test, and there were 1347 VPI who were analyzed from external test cohort.The rest of VPI were separated into the balanced model developmental set for which the numbers were 846 and 696 VPI in the CRegres, and MRegres for 6-month CA, and 532 and 660 VPI in the CDelay, and MDelay for 12-month CA, respectively.The mean GA in the original cohort and independent and external tests were 28, 27.9, and 27.9 weeks, respectively.There was no significant difference in the NDI rates between the original and external test cohorts for all four models, with the rates being slightly higher in the external test cohort (CDelay: 16.0% vs 15.3%, p = 0.56; MDelay: 20.4% vs 18.5%, p = 0.15; CRegres: 26.1% vs 23.7%, p = 0.09; MRegres: 21.4% vs 19.3%, p = 0.12).The NDI rate, z-score distribution of BBW, GA, and sex in the original cohort, balanced development model, and independent and external tests are shown in Table 1.

The performance of evolutionary learning and other methods in original cohort
The results of different models in validation and independent test were in Table 2.
The AUC of RF with all-features-in methods in the independent test at 24  of each selected variable on the predicted outcome are listed in Table 3.

External test validation
Among two different feature selection strategies, RF had the highest AUC in the all-features method, and EL-NDI had the highest AUC in the coarse-to-fine selection method.The AUC of RF with all-features-in methods in the external cohort at 24

Discussion
Previous NDI prediction models had limited sample sizes and often used black box models without clearly explaining model performance [9,10,30].In this large national sample of VPI, EL-NDI models utilized fewer predictors (4 and 10) with similar AUC compared with RF and lasso with All-features-in methods, specifically in external validation cohort.The neurodevelopment prediction models estimated and compared in this study were developed at the visit level, which might allow the physician to identify which individuals are at risk as well as worsen in the future.
It is difficult to compare the NDI performance between different studies because of the variety of NDI definitions and target groups [9,10].External validation of the Neonatal Research Network (NRN) using the estimation of five risk factors (GA at birth, exposure versus no exposure to antenatal corticosteroids, singleton versus multiple gestation, gender, and birth weight)-one of the most widely-used risk modelsshowed AUCs were 0.64 and 0.71 for death and severe NDI [31].Ambalavanan et al. used 21 variables to reach AUCs of 0.66 and 0.75 for mental and psychomotor development index, respectively, at 12 to 18 months of age and showed that neural network was not superior to the logic regression model [32].In a recent study conducted by Li et al., a machine learning prediction model based on perinatal factors, specifically utilizing SVM methodology, was found to outperform other modeling techniques such as multivariate LR, RF, and neural network analysis [7].Notably, the SVM model in Li et al. 's study achieved an AUC of 0.7 during an independent test involving 78 very preterm infants (VPI) for composite NDI outcome, including moderate to severe cerebral palsy, cognitive or motor scores below 2 standard deviations from the norm, bilateral hearing impairment necessitating hearing aids, or bilateral blindness [7].Conversely, our EL-NDI model excelled during a comprehensive external validation test, encompassing small sets of perinatal and post-NICU data.It emphasizes more nuanced and specific NDI outcomes, demonstrating its potential to provide more precise prognostic information.Neuroimaging findings in preterm infants can potentially serve as a predictive indicator of adverse neurodevelopmental outcomes [33].Moeskops et al. showed that the SVM method identified the change of brain MRIs of PMA between 30 and 40 weeks and reached AUCs of 0.80 and 0.85 for cognitive and motor BSIDIII composite scores < 85 at 2-3 years CA, respectively [34].The neural network method reported a 100% positive predictive value and a 90.6% negative predictive value for NDI at 1 year of age using advanced brain MRIs at term-equivalent age (TEA) [35].A prognostic study for NDI based on Denver screen test II with 109 extremely preterm infants using multimodal model combining electroencephalography, brain structure information, early postnatal morbidities, and perinatal factors revealed high AUCs (0.91, 95% CI, 86.4-97.0%)and demonstrated the potential of the brain functional information for prediction model [8].The issue of overfitting and small sample sizes still needs to be addressed, regardless of whether the predictive models are based on brain function, MRI, or risk factor modeling [9,30].
Previous investigations of cognitive and motor regression models have mostly focused on grouping and risk factors [21].Although the development of brain trajectories correlates with future functional outcomes [36], the correlation between the degree of BSID score regression and future outcome is still unknown.However, our aim was to detect dynamic changes as early as possible and provide an opportunity to discuss the best followup strategy.To the best of our knowledge, this is the first prediction model for Bayley score decline between two time points for VPI.We identify the individual impacts of each risk factor from our models to avoid black box models, which would be useful in routine clinical practice [9].Most determinants associated with the variables in the four models align closely with findings from prior research on adverse NDI outcomes, including factors such as paternal education, gastrointestinal surgery, and duration of mechanical ventilation, [7,37,38].Among anthropometric measurements in the dataset, only the BL and HC at 6 months CA were used in the CDelay.Although there is small amount of evidence to suggest that poor postnatal growth after discharge is associated with NDI in preterm infants [39,40], the literature raises concerns regarding the efficacy of a solitary anthropometric measure, such as BL or HC, as a direct predictor of NDI in children [41].Even though previous studies have demonstrated that very low birth weight preterm infants face various complications that may hinder their ability to achieve catch-up growth and normal neurodevelopment [41], our study explored the correlation between anthropometric measurements and other risk factors in preterm infants, particularly concerning predicting neurodevelopmental outcomes.Prolonged duration of mechanical ventilation was significantly inversely associated with NDI and brain development [42].Surprisingly, IPPV days in the CDelay model and PMA at discharge in the MDelay model  were protective factors, based on the EL-LR model.A retrospective study of factors influencing the attendance of early intervention in Iran showed that length of stay (LOS) in NICU is a primary factor that affects attendance [43].A study of very preterm in Korea found that NICU graduates who adhered to LFUP had more severe morbidities during their NICU stay and a higher PMA at discharge [44].Therefore, the IPPV days and PMA at discharge in CDelay and MDelay models may be associated with adherence to LFUP and early intervention, which help identify and treat health problems early in NICU graduates to improve outcomes.Subsequent research should investigate the impact of these factors on parental behavior and their consequences for these children.In regression models at 6-month CA, higher BSIDIII score in both cognitive and motor models was indicated as a risk factor.The severity of the child's NDI was directly proportional to the lower BSIDIII score at 6 months of CA.Thus, the BSIDIII scores in VPI with more severe NDI may exhibit less variability than their counterparts [21,23].
Despite our innovative approach, EL-NDI for CDelay still requires the inclusion of more variables to maximize accuracy compared to the other three EL-NDI models, underscoring the inherent complexity of cognitive function prediction.The difference between original and external cohorts in our study, such as a higher BSID score in two checkpoints, lower PDA ligation rate, and an increase in the proportion of lower Z scores for anthropometric measurements in the external cohort, may result from improved survival rates, care strategy change, and higher follow-up rates among very preterm infants in different periods [45].Predictive models are built upon historical data and aim to make forecasts based on past knowledge.Considering the advancement in the care of VPI, this result emphasizes the limitations of existing risk factor models encompassing all parameters for predicting NDI outcomes in this field.Consequently, it becomes evident that continuous model updates are imperative to adapt to the everevolving landscape of medical advancements [46].Including individuals' current status and postnatal data within our visit-based neurodevelopment predictive modeling strategy represents a notable strength of our study.Increasing the adherence rate in follow-up has always been one of the challenges in the care of highrisk newborns, particularly premature infants, to achieve early intervention.Including postnatal data in our prediction model offers substantive proof, assuring parents that our scrutiny extends beyond historical considerations.We are equally vested in monitoring their child's present developmental trajectory and, more crucially, comprehending how these combined elements shape the child's prospective welfare.This approach provides a comprehensive understanding of the factors influencing neurodevelopmental outcomes and serves as the cornerstone for better decision-making between physicians and families during LFUP.We are the first prediction models for NDI in VPI with machine learning methods and external validation.Our research in developing such predictive models provides the foundation for future investigations.Within the comprehensive research framework, we have gained insights into the potential of machine learning and have recognized the limitations of current risk factor-based models in this field.Unlike conventional approaches, the EL-NDI method incorporates a modelspecific signature, thereby preserving predictive accuracy across distinct cohort periods by effectively mitigating the influence of insignificant features.The integration of AI-based tools into clinical practice remains a paramount concern.AI products in clinical settings, especially those related to medical imaging, are significantly influenced by data quality [47].We utilize routinely collected data and leverage the models to minimize parameters with reporting performance metrics, thus achieving the highest level of accuracy, which might promote ease of use and consistency in utilization.Research has shown that even before fully certifying the model's functionalities, a convenient and effective tool can change the workflow of clinical professionals and ultimately optimize patient outcomes [48].

Limitations
First, we carefully select predictors with strong linear correlations to optimize model efficiency, emphasizing the importance of interpretability in model development [29].However, this coarse-to-fine feature selection might need to include the essential features that have nonlinear relationships with the outcomes and interfere with enhancing the accuracy of the model [9].Furthermore, the available data does not include information on nonmedical practices in NICU and the child's early environment, which are known to promote brain growth and neurodevelopment [49,50].Therefore, future studies will need to investigate these factors in conjunction with machine learning methods for NDI outcomes.Second, all models picked up the BSIDIII score as variables in prediction models for best performance.These models would be limited during follow-up because of a lack of trained BSID evaluators and limited budgets [51].In our investigation, we encountered a constraint in the dataset of infant cerebral information, which solely consisted of brain sonography results from the TPFN dataset.This limitation could potentially impact the predictive efficacy of our models, as the inclusion of both morphological and functional brain data has become a recognized practice in neurocritical care for premature infants [52].Research has demonstrated that integrating MRI or electroencephalography data with known risk factors can substantially enhance the predictive precision concerning NDI outcomes in preterm infants [8,35].Further prediction model studies in this field should focus on these data with large cohort validation.Fourth, despite conducting external validation, it is essential to note that the primary composition of Taiwan's national population is East Asian.We could not test the capacity of EL-NDI on other populations.In the future, we will seek opportunities to investigate the transportability of EL-NDI.

Conclusions
Our study demonstrated good performance of evolutionary learning models with fewer variables for cognitive and motor neurodevelopmental models at 6-and 12-month CA, respectively, in predicting NDI outcomes at 24 months CA.With a qualified assessment under an evaluation framework, our models would be helpful for targeted surveillance and optimal management of clinical progress in VPI and their families, promoting better decision-making.Further research is needed to explore the impact of these models on the attendance of LFUP in very preterm infants.

Table 1
NDI had the highest AUC in CDelay, MDelay, and CRegres in the independent test, but RF had the highest AUC in MRegres with 75 variables (Table2).The CDelay encompassed 10 variables: motor BSIDIII score at 12 months, cognitive BSIDIII scores at 12 months, abdominal surgery, intermittent positive pressure ventilation (IPPV) days, pH in first-time blood gas, oxygenation supply≧ 40%, head circumstance (HC) at 6 months CA, maternal education ≦ 12 years, body length (BL) at 6 months CA, and hemodynamic significant PDA.The MDelay encompassed 4 variables: motor BSIDIII scores at 12 months, cognitive BSIDIII scores at 12 months, NICU days, and post-menstrual age (PMA) while discharge.The CRegres encompassed 4 variables: cognitive BSIDIII scores at 6 months, motor BSIDIII scores at 6 months, maternal MgSO4 use, and parental education ≦ at 12 years.The MRegres encompassed 4 variables: motor BSIDIII scores at 6 months, antenatal steroid use, HC at admission, and prolonged rupture of membranes.The formulas of EL-LR to help interpret the influence Gestational age, gender, birth body weight, and outcome in different models and sets

Table 2
Performance of different methods in original cohort

Table 3
EL-LR formula based on variables selected by the EL-NDI

Table 4
Performance of EL-NDI and RF in external validation cohort (n = 1347)